from config import  config
import wandb
run = wandb.init(project="lightning_logs", job_type='job-type')
# run = wandb.init(project='wandb_demo', id= model.run_id, resume='must')
# save dataset
arti_dataset = wandb.Artifact('mnist原始', type='dataset')
arti_dataset.add_dir('../data/')
wandb.log_artifact(arti_dataset)


arti_code = wandb.Artifact('py', type='code')
arti_code.add_file('./train_lightning.py')
wandb.log_artifact(arti_code)

# save model

arti_model = wandb.Artifact('cnn', type='model')
arti_model.add_file(config.ckpt_path)
wandb.log_artifact(arti_model)

wandb.finish()
